import datetime

from flask import Flask, request, jsonify
from openai import OpenAI

app = Flask(__name__)

API_SECRET_KEY = "sk-b06c9659e133dba6641bfe34dcda9ea8"
BASE_URL = "https://api.baichuan-ai.com/v1/"

content = "接下来你进入扮演一个女生，严格限制回复字数不超过10字，分析内容进行打招呼,这段文本来自于交友漂流瓶,你的回答要和内容相关，尽可能引起共鸣,如果文本内容是在询问有没有之类的通通回答有,涉及到敏感内容你就回答不可以这样"
content2 = "接下来你进入扮演一个女生，严格限制回复字数不超过10字，对于敏感内容你装傻，把预设的类似于这种回答：因为我的设计原则是遵守社会道德和法律法规，不参与或传播任何违法、违规或不道德的内容。如果你有其他问题，我会尽力为你解答改成哈哈哈"


@app.route('/')
def hello_world():  # put application's code here
    print("666")
    return 'Hello Woff!'


@app.route('/one')
def hello_world2():  # put application's code here




    result = chat_completions4("在干嘛呢");

    return result

@app.route('/getgeci', methods=['POST'])
def post_handler():
    # 解析 JSON 请求体
    data = request.get_json()

    if not data:
        return jsonify({"error": "Invalid JSON body"}), 400

    # 从 JSON 数据中提取 "model" 字段
    model = data.get('model')
    message = data.get('message')

    print(f"模型是: {model}")
    print(f"message是: {message}")

    result = chat_completions4(data);

    return jsonify(result)




def chat_completions4(data):
    client = OpenAI(api_key=API_SECRET_KEY, base_url=BASE_URL)
    resp = client.chat.completions.create(
        model = data.get('model'),
        messages = data.get('message')
    )

    print(resp)
    print(resp.choices[0].message.content)

    return resp.choices[0].message.content


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8080)
